package cn.chencaiju.algorithm;

import java.util.ArrayList;
import java.util.List;

import org.junit.Test;

public class MatrixReshape {
	public int[][] matrixReshape(int[][] nums, int r, int c) {
		if(nums.length*nums[0].length!=r*c) {
			return nums;
		}
		List<Integer> list=new ArrayList<>();
		for(int i=0;i<nums.length;i++) {
			for(int j=0;j<nums[i].length;j++) {
				list.add(nums[i][j]);
			}
		}
		int index=0;
		int[][] result=new int[r][c];
		for(int i=0;i<r;i++) {
			for(int j=0;j<c;j++) {
				result[i][j]=list.get(index++);
			}
		}
		return result;
	}
	@Test public void testMatrixReshape() {
		int[][] nums= {{2,3},{22,11},{22,44}};
		int[][]	result=matrixReshape(nums,2,3);
		for(int i=0;i<result.length;i++) {
			for(int j=0;j<result[i].length;j++) {
				System.out.print(result[i][j]+" ");
			}
			System.out.println();
		}
	}
}
